/**
 * Copyright (C) <2021>  <chen junwen>
 * <p>
 * This program is free software: you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * <p>
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
 * even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 * <p>
 * You should have received a copy of the GNU General Public License along with this program.  If
 * not, see <http://www.gnu.org/licenses/>.
 */
package io.mycat.calcite.rewriter;

import io.mycat.calcite.logical.MycatView;
import io.mycat.calcite.physical.MycatGather;
import io.mycat.calcite.physical.MycatMergeSort;
import org.apache.calcite.rel.RelNode;
import org.apache.calcite.rel.RelShuttleImpl;

public class GatherRewriter  extends RelShuttleImpl {

    @Override
    public RelNode visit(RelNode other) {
        RelNode relNode = visit(other);
        if (relNode instanceof MycatView) {
           return MycatGather.create(relNode);
        }else if (relNode instanceof MycatMergeSort) {
            return MycatGather.create(relNode);
        }
        return super.visit(other);
    }
}
